package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/xuri/excelize/v2"
)

func main2() {
	// 指定目录
	dir := `C:\Users\admin\Desktop\南部新城\环卫\s` // 修改为实际目录路径

	// 遍历目录下的所有文件
	files, err := os.ReadDir(dir)
	if err != nil {
		log.Fatalf("无法读取目录: %v", err)
	}

	// 遍历文件
	for _, file := range files {
		// 只处理 .xlsx 文件
		if !file.IsDir() && strings.HasSuffix(file.Name(), ".xlsx") {
			// 获取文件路径
			filePath := filepath.Join(dir, file.Name())

			// 打开 Excel 文件
			f, err := excelize.OpenFile(filePath)
			if err != nil {
				log.Printf("无法打开文件 %s: %v", file.Name(), err)
				continue
			}

			// 获取文件名中的时间前缀
			timePrefix := strings.Split(file.Name(), "-")[0]
			if len(timePrefix) != 8 {
				log.Printf("文件名 %s 的时间前缀格式不正确", file.Name())
				continue
			}

			// 将 E2 单元格的值修改为时间前缀
			sheet := "服务器检查记录表" // 默认工作表名称，可根据实际情况修改
			cell := "E2"
			f.SetCellValue(sheet, cell, timePrefix)

			// 保存文件
			if err := f.Save(); err != nil {
				log.Printf("无法保存文件 %s: %v", file.Name(), err)
				continue
			}

			fmt.Printf("已更新文件: %s\n", file.Name())
		}
	}

	fmt.Println("所有文件处理完成！")
}
